Node / meshtastic / Meshtastic-Android / files / core / data / src / commonTest / kotlin / org / meshtastic / core / data / ai / RateLimiterTest.kt

Displaying Raw • Download

T8b949e/*

* Copyright (c) 2026 Meshtastic LLC

*

* This program is free software: you can redistribute it and/or modify

* it under the terms of the GNU General Public License as published by

* the Free Software Foundation, either version 3 of the License, or

* (at your option) any later version.

*

* This program is distributed in the hope that it will be useful,

* but WITHOUT ANY WARRANTY; without even the implied warranty of

* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

* GNU General Public License for more details.

*

* You should have received a copy of the GNU General Public License

* along with this program. If not, see <https://www.gnu.org/licenses/>.

*/

Tff7b72package T7ee787org.meshtastic.core.data.ai

Tff7b72import T7ee787kotlinx.coroutines.test.runTest

Tff7b72import T7ee787kotlin.test.Test

Tff7b72import T7ee787kotlin.test.assertEquals

Tff7b72import T7ee787kotlin.test.assertIs

Tff7b72import T7ee787kotlin.time.Clock

Tff7b72import T7ee787kotlin.time.Duration.Companion.seconds

Tff7b72import T7ee787kotlin.time.Instant

Tff7b72class T56d364RateLimiterTest Tb4b4b4{

Tf0883e@Test

Tff7b72fun Td2a8ffpermits_calls_under_limitTb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3clock Tff7b72= Te6edf3FakeClockTb4b4b4(Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff0T79c0ff0Tb4b4b4)Tb4b4b4)

Tff7b72val Te6edf3rateLimiter Tff7b72= Te6edf3RateLimiterTb4b4b4(Te6edf3clockTb4b4b4)

Te6edf3repeatTb4b4b4(Te6edf3RateLimiterTb4b4b4.Te6edf3MAX_CALLSTb4b4b4) Tb4b4b4{ Te6edf3assertIsTff7b72<Te6edf3RateLimitResultTb4b4b4.Te6edf3PermittedTff7b72>Tb4b4b4(Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4)Tb4b4b4) Tb4b4b4}

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ffrejects_calls_over_limitTb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3clock Tff7b72= Te6edf3FakeClockTb4b4b4(Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff0T79c0ff0Tb4b4b4)Tb4b4b4)

Tff7b72val Te6edf3rateLimiter Tff7b72= Te6edf3RateLimiterTb4b4b4(Te6edf3clockTb4b4b4)

T8b949e// Exhaust the limit

Te6edf3repeatTb4b4b4(Te6edf3RateLimiterTb4b4b4.Te6edf3MAX_CALLSTb4b4b4) Tb4b4b4{ Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4) Tb4b4b4}

Tff7b72val Te6edf3result Tff7b72= Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4)

Te6edf3assertIsTff7b72<Te6edf3RateLimitResultTb4b4b4.Te6edf3LimitedTff7b72>Tb4b4b4(Te6edf3resultTb4b4b4)

Te6edf3assertEqualsTb4b4b4(T79c0ff6T79c0ff1Tb4b4b4, Te6edf3resultTb4b4b4.Te6edf3retryAfterSecondsTb4b4b4) T8b949e// full window remaining + 1

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ffpermits_after_window_expiresTb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3clock Tff7b72= Te6edf3FakeClockTb4b4b4(Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff0T79c0ff0Tb4b4b4)Tb4b4b4)

Tff7b72val Te6edf3rateLimiter Tff7b72= Te6edf3RateLimiterTb4b4b4(Te6edf3clockTb4b4b4)

T8b949e// Exhaust the limit

Te6edf3repeatTb4b4b4(Te6edf3RateLimiterTb4b4b4.Te6edf3MAX_CALLSTb4b4b4) Tb4b4b4{ Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4) Tb4b4b4}

T8b949e// Advance past the window

Te6edf3clockTb4b4b4.Te6edf3currentTime Tff7b72= Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff0T79c0ff0Tb4b4b4) Tff7b72+ Te6edf3RateLimiterTb4b4b4.Te6edf3WINDOW_DURATION Tff7b72+ T79c0ff1.Te6edf3seconds

Te6edf3assertIsTff7b72<Te6edf3RateLimitResultTb4b4b4.Te6edf3PermittedTff7b72>Tb4b4b4(Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4)Tb4b4b4)

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ffsliding_window_evicts_oldest_entryTb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3clock Tff7b72= Te6edf3FakeClockTb4b4b4(Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff0T79c0ff0Tb4b4b4)Tb4b4b4)

Tff7b72val Te6edf3rateLimiter Tff7b72= Te6edf3RateLimiterTb4b4b4(Te6edf3clockTb4b4b4)

T8b949e// Fill the window with calls 10 seconds apart

Te6edf3repeatTb4b4b4(Te6edf3RateLimiterTb4b4b4.Te6edf3MAX_CALLSTb4b4b4) Tb4b4b4{ Te6edf3i Tff7b72-Tff7b72>

Te6edf3clockTb4b4b4.Te6edf3currentTime Tff7b72= Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff0T79c0ff0L Tff7b72+ Te6edf3i Tff7b72* T79c0ff1T79c0ff0Tb4b4b4)

Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4)

Tb4b4b4}

T8b949e// At t=1050, first call (t=1000) is still in window (threshold is t=990)

Te6edf3clockTb4b4b4.Te6edf3currentTime Tff7b72= Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff5T79c0ff0Tb4b4b4)

Te6edf3assertIsTff7b72<Te6edf3RateLimitResultTb4b4b4.Te6edf3LimitedTff7b72>Tb4b4b4(Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4)Tb4b4b4)

T8b949e// At t=1061 — first call (t=1000) should have expired from window

Te6edf3clockTb4b4b4.Te6edf3currentTime Tff7b72= Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff6T79c0ff1Tb4b4b4)

Te6edf3assertIsTff7b72<Te6edf3RateLimitResultTb4b4b4.Te6edf3PermittedTff7b72>Tb4b4b4(Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4)Tb4b4b4)

Tb4b4b4}

Tf0883e@Test

Tff7b72fun Td2a8ffretry_after_is_accurateTb4b4b4(Tb4b4b4) Tff7b72= Te6edf3runTest Tb4b4b4{

Tff7b72val Te6edf3clock Tff7b72= Te6edf3FakeClockTb4b4b4(Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff0T79c0ff0Tb4b4b4)Tb4b4b4)

Tff7b72val Te6edf3rateLimiter Tff7b72= Te6edf3RateLimiterTb4b4b4(Te6edf3clockTb4b4b4)

T8b949e// All calls at t=1000

Te6edf3repeatTb4b4b4(Te6edf3RateLimiterTb4b4b4.Te6edf3MAX_CALLSTb4b4b4) Tb4b4b4{ Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4) Tb4b4b4}

T8b949e// Check at t=1030 (halfway through window)

Te6edf3clockTb4b4b4.Te6edf3currentTime Tff7b72= Te6edf3InstantTb4b4b4.Te6edf3fromEpochSecondsTb4b4b4(T79c0ff1T79c0ff0T79c0ff3T79c0ff0Tb4b4b4)

Tff7b72val Te6edf3result Tff7b72= Te6edf3rateLimiterTb4b4b4.Te6edf3tryAcquireTb4b4b4(Tb4b4b4)

Te6edf3assertIsTff7b72<Te6edf3RateLimitResultTb4b4b4.Te6edf3LimitedTff7b72>Tb4b4b4(Te6edf3resultTb4b4b4)

T8b949e// Oldest at t=1000, expires at t=1060, now is t=1030, so retryAfter = 31

Te6edf3assertEqualsTb4b4b4(T79c0ff3T79c0ff1Tb4b4b4, Te6edf3resultTb4b4b4.Te6edf3retryAfterSecondsTb4b4b4)

Tb4b4b4}

Tb4b4b4}

T8b949e/** Simple fake Clock for testing. */

Tff7b72private Tff7b72class T56d364FakeClockTb4b4b4(Tff7b72var Te6edf3currentTimeTb4b4b4: Te6edf3InstantTb4b4b4) Tb4b4b4: Te6edf3Clock Tb4b4b4{

Tff7b72override Tff7b72fun Td2a8ffnowTb4b4b4(Tb4b4b4)Tb4b4b4: Te6edf3Instant Tff7b72= Te6edf3currentTime

Tb4b4b4}